#ifndef __M_BUFFER_HPP__
#define __M_BUFFER_HPP__

//实现异步日志缓冲区
#include <iostream>
#include <vector>
#include <cassert>
#include "util.hpp"

namespace mylog
{
    #define DEFAULT_BUFFER_SIZE (100 * 1024 * 1024)
    #define THREASHOLD_BUFFER_SIZE (80 * 1024 * 1024) //阈值
    #define INCREMENT_BUFFER_SIZE (10 * 1024 * 1024)   //线性增长
    class Buffer
    {
    private:
        void moveWriter(size_t len)
        {
            assert(len + _writer_idx <= _buffer.size());
            _writer_idx += len;
        }
        //确保缓冲区有足够的空间容纳len长度的数据
        void ensureEnoughSize(size_t len)
        {
            if(len < writeAbleSize())
                return;
            size_t new_size = 0;
            if(_buffer.size () < THREASHOLD_BUFFER_SIZE)
            {
                new_size = _buffer.size() * 2 + len; //小于阈值则翻倍增长
            }
            else
            {
                //大于阈值则线性增长
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len;
            }
            _buffer.resize(new_size);
        }
    public:
        Buffer()
            :_buffer(DEFAULT_BUFFER_SIZE),
            _reader_idx(0),
            _writer_idx(0)
        {}
        //写入
        void push(const char* data, size_t len)
        {
            //缓冲区剩余空间不够的情况：1、扩容； 2、阻塞/返回false
            //扩容实际上不适用，资源需要限制，不宜无限扩容
            // //1、固定大小，则直接返回
            // if(len > writeAbleSize()) 
            //     return;
            //2、扩容，则返回
            ensureEnoughSize(len);
            //将数据拷贝进入缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            //将当前写入位置向后偏移
            moveWriter(len);
        }
        //可读数据的起始地址
        const char* begin()
        {
            return &_buffer[_reader_idx];
        } 
        //可读数据的长度
        size_t readAbleSize()
        {
            //因为当前实现的缓冲区设计思想是双缓冲区，处理完就交换，所以不存在空间循环使用的情况
            return _writer_idx - _reader_idx;
        }
        size_t writeAbleSize()
        {
            //对于选择扩容方案来说，不存在可写空间大小，因为总是可写
            //因此这个接口仅仅针对固定大小缓冲区提供
            return _buffer.size() - _writer_idx;
        }
        //读写指针进行向后偏移操作
        void moveReader(size_t len)
        {
            assert(len <= readAbleSize());
            _reader_idx += len;
        }
        //重置读写位置，初始化缓冲区
        void reset()
        {
            _reader_idx = 0;
            _writer_idx = 0;
        }
        //对Buffer进行交换
        void swap(Buffer& buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }
        //判空
        bool empty()
        {
            return _reader_idx == _writer_idx;
        }
    private:
        std::vector<char> _buffer;
        size_t _reader_idx;
        size_t _writer_idx; 
    };
}

#endif